
package com.wjk.kylin.core.utils;

import cn.hutool.core.util.IdUtil;
import com.aliyun.oss.OSS;
import com.aliyun.oss.OSSClientBuilder;
import com.wjk.kylin.config.properties.AliCloudOSSProperties;
import com.wjk.kylin.core.exception.BusinessException;
import org.apache.commons.lang3.StringUtils;

import java.io.ByteArrayInputStream;
import java.io.InputStream;
import java.util.Date;

/**
 * 阿里云存储
 *
 * @author Mark sunlightcs@gmail.com
 */
public class AliCloudOSSUtils {
    private OSS ossClient;
    public AliCloudOSSUtils(){
        //初始化
        init();
    }

    private void init(){
        ossClient = new OSSClientBuilder().build(AliCloudOSSProperties.END_POIND, AliCloudOSSProperties.ACCESS_KEY_ID, AliCloudOSSProperties.ACCESS_KEY_SECRET);
    }

    /**
     * 文件路径
     * @param prefix 前缀
     * @param suffix 后缀
     * @return 返回上传路径
     */
    public String getPath(String prefix, String suffix) {
        //文件路径
        String path = DateUtils.formatDate(new Date(), "yyyy/MM/dd") + "/" +  IdUtil.simpleUUID();

        if(StringUtils.isNotBlank(prefix)){
            path = prefix + "/" + path;
        }

        return path + suffix;
    }

    /**
     * 文件上传
     * @param data    文件字节数组
     * @param path    文件路径，包含文件名
     * @return        返回http地址
     */
    public String upload(byte[] data, String path) {
        return upload(new ByteArrayInputStream(data), path);
    }

    /**
     * 文件上传
     * @param inputStream   字节流
     * @param path          文件路径，包含文件名
     * @return              返回http地址
     */
    public String upload(InputStream inputStream, String path) {
        try {
            ossClient.putObject(AliCloudOSSProperties.BUCKET_NAME, path, inputStream);
        } catch (Exception e){
            throw new BusinessException("上传文件失败，请检查配置信息");
        } finally {
            // 关闭OSSClient。
            ossClient.shutdown();
        }

        return AliCloudOSSProperties.DOMAIN + "/" + path;
    }

    /**
     * 文件上传
     * @param data          文件字节数组
     * @param suffix        后缀
     * @return              返回http地址
     */
    public String uploadSuffix(byte[] data, String suffix) {
        return upload(data, getPath(AliCloudOSSProperties.PREFIX, suffix));
    }

    /**
     * 文件上传
     * @param inputStream  字节流
     * @param suffix       后缀
     * @return             返回http地址
     */
    public String uploadSuffix(InputStream inputStream, String suffix) {
        return upload(inputStream, getPath(AliCloudOSSProperties.PREFIX, suffix));
    }
}
